Skip to content

brute_force_prefiltered test warp boundary but with less input - #2579

Open
robertmaynard wants to merge 5 commits into
NVIDIA:mainfrom
robertmaynard:fea/improve_neighbors_brute_force_prefiltered_test_time
Open

robertmaynard wants to merge 5 commits into
NVIDIA:mainfrom
robertmaynard:fea/improve_neighbors_brute_force_prefiltered_test_time

Conversation

@robertmaynard

Copy link
Copy Markdown
Contributor

For the longest running test items in brute_force_prefiltered.cu we can do a tweaks to improve runtime:

  • Use smaller input dim values which aren't a multiple of the vector width to still validate serial trailing loads
  • Increase the sparsity but keep about the kSddmmMaxSelectivity threshold

These combine together to remove 20 minutes from NEIGHBORS_TEST .

@robertmaynard
robertmaynard requested a review from a team as a code owner September 9, 2026 19:50
@robertmaynard robertmaynard added improvement Improves an existing functionality non-breaking Introduces a non-breaking change ci labels Sep 9, 2026

@tfeher tfeher left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks @robertmaynard!

@robertmaynard
robertmaynard force-pushed the fea/improve_neighbors_brute_force_prefiltered_test_time branch from b9bac9a to 1179554 Compare September 12, 2026 00:03
The brute_force_prefiltered previous changes back to `205[0,1,2]` didn't fix the issue so increase the density arg.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ci improvement Improves an existing functionality non-breaking Introduces a non-breaking change

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants